package org.algorithm.熊猫.回溯算法;

import org.junit.Test;

import java.util.LinkedList;
import java.util.List;

/**
 * 找出所有相加之和为n 的k个数的组合，且满足下列条件：
 * <p>
 * 只使用数字1到9
 * 每个数字最多使用一次
 * 返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。
 * 216. 组合总和 III
 * 链接：https://leetcode-cn.com/problems/combination-sum-iii
 */
public class $4月25日$组合总和3 {

    @Test
    public void test() {
        System.out.println(combinationSum3(3, 9));
    }


    public List<List<Integer>> combinationSum3(int k, int n) {
        int[] arr = new int[9];
        for (int i = 0; i < 9; i++) {
            arr[i] = i + 1;
        }
        backtrack(arr, 0, k, n);
        return res;
    }

    List<List<Integer>> res = new LinkedList<>();
    // 记录回溯算法的递归路径
    LinkedList<Integer> track = new LinkedList<>();


    int count = 0;

    // 回溯算法核⼼函数，遍历⼦集问题的回溯树
    void backtrack(int[] nums, int start, int k, int n) {

        if (track.size() > k) {
            return;
        }
        if (track.size() == k && n == count) {
            res.add(new LinkedList<>(track));
            return;
        }

        // 回溯算法标准框架
        for (int i = start; i < nums.length; i++) {
            // 做选择
            count += nums[i];
            track.addLast(nums[i]);
            // 通过 start 参数控制树枝的遍历，避免产⽣重复的⼦集
            backtrack(nums, i + 1, k, n);
            // 撤销选择
            count -= nums[i];
            track.removeLast();
        }
    }
}
